"""
虚拟gpu  :  把一块gpu分成几块虚拟的gpu  

"""

import tensorflow as tf 

tf.debugging.set_log_device_placement(True)

gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_virtual_device_configuration(
    gpus[0],  # 指定虚拟化第一块gpu
    [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048),
    tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)]  # 列表的方式传入虚拟化的数目和大小
)
print(len(gpus))  # 物理设备4块

logical_gpus = tf.config.experimental.list_logical_devices('CPU')
print(len(logical_gpus))  #　逻辑设备5块 如果设置了虚拟化的这块gpu可见 那就是2块